# download.py —— 兼容旧版 huggingface_hub
from huggingface_hub import snapshot_download
import os

# 设置镜像（建议在命令行中设置，或在这里用 os.environ）
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

print("开始下载模型...")

try:
    snapshot_download(
        repo_id="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
        local_dir="./models/sentence-transformer",
        force_download=True,
        resume_download=True,
        # max_retries=5,   # 如果版本不支持，注释掉
        # timeout=600,     # 如果版本不支持，注释掉
        token=False,
    )
    print("✅ 下载完成！模型保存在: ./models/sentence-transformer")
except TypeError as e:
    if "max_retries" in str(e):
        print("⚠️ 你的 huggingface_hub 版本较旧，不支持 max_retries/timeout，正在使用默认参数重试...")
        snapshot_download(
            repo_id="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
            local_dir="./models/sentence-transformer",
            force_download=True,
            resume_download=True,
            token=False,
        )
        print("✅ 下载完成！")
    else:
        raise e